package com.jdj.framework.core.base.utils;

import java.util.Collection;
import java.util.Iterator;
import java.util.List;
import org.apache.commons.lang3.StringUtils;

/**
 * Collections工具集.
 * @author caijinbang
 * @date 2019-04-13 15:23
 */
public class CollectionUtils extends org.apache.commons.collections.CollectionUtils{

    /**
     * 转换Collection所有元素(通过toString())为String, 中间以 separator分隔。
     */
    @SuppressWarnings("rawtypes")
    public static String convertToString(final Collection collection, final String separator) {
        return StringUtils.join(collection, separator);
    }

    /**
     * 转换Collection所有元素(通过toString())为String,
     * 每个元素的前面加入prefix，后面加入postfix，如<div>mymessage</div>。
     */
    @SuppressWarnings("rawtypes")
    public static String convertToString(final Collection collection, final String prefix, final String postfix) {
        StringBuilder builder = new StringBuilder();
        for (Object o : collection) {
            builder.append(prefix).append(o).append(postfix);
        }
        return builder.toString();
    }

    /**
     * 取得Collection的第一个元素，如果collection为空返回null.
     */
    public static <T> T getFirst(Collection<T> collection) {
        if (isEmpty(collection)) {
            return null;
        }

        return collection.iterator().next();
    }

    /**
     * 获取Collection的最后一个元素 ，如果collection为空返回null.
     */
    public static <T> T getLast(Collection<T> collection) {
        if (isEmpty(collection)) {
            return null;
        }

        // 当类型为List时，直接取得最后一个元素 。
        if (collection instanceof List) {
            List<T> list = (List<T>) collection;
            return list.get(list.size() - 1);
        }

        // 其他类型通过iterator滚动到最后一个元素.
        Iterator<T> iterator = collection.iterator();
        while (true) {
            T current = iterator.next();
            if (!iterator.hasNext()) {
                return current;
            }
        }
    }

    /**
     * 清除Collection中的null元素
     *
     * @param collection 需要清除null元素的Collection
     * @return 已经清除null元素的Collection
     */
    public static Collection<?> clearNullElement(Collection<?> collection) {
        if (collection != null) {
            Iterator<?> it = collection.iterator();
            while (it.hasNext()) {
                if (it.next() == null) {
                    it.remove();
                }
            }
        }
        return collection;
    }

    /**
     * 清除Collection中的特定的元素
     *
     * @param collection 需要清除特定元素的Collection
     * @param element    需要被清除的特定元素
     * @return 已经清除特定元素的Collection
     */
    public static Collection<?> clearElement(Collection<?> collection, Object element) {
        if ((collection != null) && (element != null)) {
            Iterator<?> it = collection.iterator();
            while (it.hasNext()) {
                if (it.next().equals(element)) {
                    it.remove();
                }
            }
        }
        return collection;
    }
}
